package uvt.map.graph.element;

import java.awt.Color;

public enum VehicleColor {
	
	TRAM(Color.yellow),
	TROLLEYBUS(Color.magenta),
	BUS(Color.blue);
	
	private Color color;
	
	VehicleColor(Color color) {
		this.color = color;
	}
	
	public static Color getTramColor() {
		return TRAM.color;
	}
	
	public static Color getTrolleybusColor() {
		return TROLLEYBUS.color;
	}
	
	public static Color getBusColor() {
		return BUS.color;
	}
	
	public static Color getColorForVehicle(String type) {
		Color color = null;
		
		if(type.equals("Tram")) {
			color = TRAM.color;
		} else if(type.equals("Trolleybus")) {
			color = TROLLEYBUS.color;
		} else if(type.equals("Bus")) {
			color = BUS.color;
		}
		
		return color;
	}

}
